/* Universidad del Valle de Guatemala
 * Mariana Castillo, 12374
 * Julio Maldonado, 12377
 * Jose Pablo Rodriguez, 12364
 * Seccion 30, Algoritmos y Estructura de Datos
 * Implementacion Treemap
 */

import java.util.HashMap;
import java.util.*;

public class TreeMapImpl implements WordSet
{
	private TreeMap<String,String>  mapa;
	
        //Se crea un TreeMap de JCF
	public TreeMapImpl(){
		mapa= new TreeMap();
	}
	
        //Se agregan los elementos al TreeMap, con su llave y valor
	public void add(Word word){
		mapa.put(word.getWord(),word.getType());
	}
	
        //Se revisa si el Treemap contiene el objeto, si lo contiene devuelve un Word
	public Word get(Word word){
        if (!(mapa.containsKey(word.getWord())))
            return null;
        return new Word(word.getWord(),mapa.get(word.getWord()));
       
	}
	
	
}